import numpy as np
import csv
from datetime import datetime
from time import strftime


def getCurTime():
    """
    get current time
    Return value of the date string format(%Y-%m-%d %H:%M:%S)
    """
    format='%Y-%m-%d %H:%M:%S'
    sdate = None
    cdate = datetime.now()
    try:
        sdate = cdate.strftime(format)
    except:
        raise ValueError
    return sdate

def build_data_list(inputCSV):
    sKey = []
    fn = inputCSV
    f = open(inputCSV)
    #ra = csv.DictReader(file(fn), dialect="excel")
    ra = csv.DictReader(f, dialect="excel")
    
    for record in ra:
        #print record[ra.fieldnames[0]], type(record[ra.fieldnames[-1]])
        for item in ra.fieldnames:
            temp = int(float(record[item]))
            sKey.append(temp)
    sKey = np.array(sKey)
    sKey.shape=(-1,len(ra.fieldnames))
    return sKey


#--------------------------------------------------------------------------
#MAIN

if __name__ == "__main__":
    print '===================================================='
    print "begin at " + getCurTime()

    filepath = 'C:/_DATA/migration_census_2009/cnty_fid_stateid.csv'
    #filepath = 'C:/_DATA/migration_census_2000/Aggregation/cnty_aggr_min_100k_stateid.csv'
    #filepath = 'C:/_DATA/migration89_08/COUNTY Migration/clean/test/US_census_pop_2011_county_id.csv'

    data = build_data_list(filepath)

    state = np.unique(data[:,-1]/1000)
    state_id = {}
    i = 0
    for item in state:
        state_id[item] = i
        i += 1

    print state
    print state_id

    
    for item in data:
        item[-1] = state_id[item[-1]/1000]

    print data[:,-1]



    np.savetxt(filepath[:-4] + '_state.csv', data, delimiter=',', fmt = '%s')
    